import torch
import  torchvision

# 方式 1
model = torch.load("../../../dataSet/model/vgg16_method1.pdh")

vgg16 = torchvision.models.vgg16(pretrained=False)
# 方式2
vgg16.load_state_dict(torch.load("../../../dataSet/model/vgg16_method2.pdh"))

print(vgg16)
